#ifndef _DRIVER_H
#define _DRIVER_H

#include <string>
#include <vector>

#include "generator.h"
#include "coder.h"

typedef std::pair<unsigned char*, unsigned> cpair;

class Driver {
  public:
    // seed is for AES, key is for srand.
    Driver(int key, int seed, std::string& fname);

    int DoEncrypt(const std::string& text,
        const std::string& bin, unsigned ammount);
    int DoDecrypt(const std::string& bin, const std::string& text);

  private:
    int _key;
    int _seed;
    Generator _gen;
    Coder _coder;
};

#endif  // _DRIVER_H
